from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score, confusion_matrix,classification_report
from sklearn.datasets import load_iris

iris=load_iris()
x=iris.data
y=iris.target

x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.2,random_state=42)

model=GaussianNB()
model.fit(x_train,y_train)
y_pred=model.predict(x_test)

print("\nConfusion Matrix:")
print(confusion_matrix(y_test, y_pred))

print("accuracy",accuracy_score(y_test,y_pred))
print("\nPrediction for new sample:", iris.target_names[y_pred][0])
